﻿
// StringEncryptionDlg.cpp: 实现文件
//

#include "pch.h"
#include "framework.h"
#include "StringEncryption.h"
#include "StringEncryptionDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialogEx
{
public:
	CAboutDlg();

// 对话框数据
#ifdef AFX_DESIGN_TIME
	enum { IDD = IDD_ABOUTBOX };
#endif

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CStringEncryptionDlg 对话框



CStringEncryptionDlg::CStringEncryptionDlg(CWnd* pParent /*=nullptr*/)
	: CDialogEx(IDD_STRINGENCRYPTION_DIALOG, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CStringEncryptionDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_EDIT_ANSI, m_Edit_Ansi);
	DDX_Control(pDX, IDC_EDIT_RESULTS, m_Edit_Results);
	DDX_Control(pDX, IDC_EDIT_STRING, m_Edit_String);
	DDX_Control(pDX, IDC_EDIT_UNICODE, m_Edit_UniCode);
}

BEGIN_MESSAGE_MAP(CStringEncryptionDlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTON_UNICODE, &CStringEncryptionDlg::OnBnClickedButtonUnicode)
	ON_BN_CLICKED(IDC_BUTTON_ANSI, &CStringEncryptionDlg::OnBnClickedButtonAnsi)
END_MESSAGE_MAP()


// CStringEncryptionDlg 消息处理程序

BOOL CStringEncryptionDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != nullptr)
	{
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码

	m_Edit_String.SetFocus();

	return FALSE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CStringEncryptionDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialogEx::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CStringEncryptionDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作区矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialogEx::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CStringEncryptionDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CStringEncryptionDlg::OnBnClickedButtonUnicode()
{
	CString InString;
	CString ResultsSting;
	CString FormatString;

	m_Edit_String.GetWindowText(InString);
	if (InString.GetLength() == 0)
	{
		MessageBox(L"请输入需要转换的字符串");
		return;
	}

	AnsiAndUniCodeToString(InString);
	
	int Xor1 = GetRand(4369, 65535);
	int Xor2 = GetRand(4369, 65535);
	int Add1 = GetRand(0, 65535);
	int Add2 = GetRand(0, 65535);
	int Add3 = GetRand(0, 65535);

	FormatString.Format(L"wchar_t ws_%ls[] = { ", InString.GetBuffer());
	ResultsSting.Append(FormatString);
	for (size_t i = 0; i < InString.GetLength(); i++)
	{
		wchar_t EncryptedValues = *(InString.GetBuffer() + i);

		EncryptedValues += Add1;
		EncryptedValues ^= Xor1;
		EncryptedValues += Add2;
		EncryptedValues ^= Xor2;
		EncryptedValues += Add3;


		FormatString.Format(L"0x%04X, ", EncryptedValues);
		ResultsSting.Append(FormatString);
	}

	ResultsSting.Append(L" 0 }; \r\n\r\n");

	FormatString.Format(L"for (size_t i = 0; i < %d; i++)\r\n", InString.GetLength());
	ResultsSting.Append(FormatString);

	FormatString.Format(L"{\r\n");
	ResultsSting.Append(FormatString);

	FormatString.Format(L"\tws_%ls[i] -= 0x%04X;\r\n", InString.GetBuffer(), Add3);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tws_%ls[i] ^= 0x%04X;\r\n", InString.GetBuffer(), Xor2);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tws_%ls[i] -= 0x%04X;\r\n", InString.GetBuffer(), Add2);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tws_%ls[i] ^= 0x%04X;\r\n", InString.GetBuffer(), Xor1);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tws_%ls[i] -= 0x%04X;\r\n", InString.GetBuffer(), Add1);
	ResultsSting.Append(FormatString);

	FormatString.Format(L"};\r\n\r\n");
	ResultsSting.Append(FormatString);

	FormatString.Format(L"UNICODE_STRING unFuncName%ls = { 0 };\r\n", InString.GetBuffer());
	ResultsSting.Append(FormatString);
	FormatString.Format(L"RtlInitUnicodeString(&unFuncName%ls, ws_%ls);\r\n", InString.GetBuffer(), InString.GetBuffer());
	ResultsSting.Append(FormatString);
	FormatString.Format(L"PUCHAR Func%ls = (PUCHAR)MmGetSystemRoutineAddress(&unFuncName%ls);\r\n", InString.GetBuffer(), InString.GetBuffer());
	ResultsSting.Append(FormatString);

	m_Edit_Results.SetWindowText(ResultsSting);
}

void CStringEncryptionDlg::OnBnClickedButtonAnsi()
{
	CString InString;
	CString ResultsSting;
	CString FormatString;

	m_Edit_String.GetWindowText(InString);
	if (InString.GetLength() == 0)
	{
		MessageBox(L"请输入需要转换的字符串");
		return;
	}

	AnsiAndUniCodeToString(InString);

	int Xor1 = GetRand(11, 255);
	int Xor2 = GetRand(11, 255);
	int Add1 = GetRand(0, 255);
	int Add2 = GetRand(0, 255);
	int Add3 = GetRand(0, 255);

	FormatString.Format(L"unsigned char as_%ls[] = { ", InString.GetBuffer());
	ResultsSting.Append(FormatString);
	for (size_t i = 0; i < InString.GetLength(); i++)
	{
		unsigned char EncryptedValues = *(InString.GetBuffer() + i);

		EncryptedValues += Add1;
		EncryptedValues ^= Xor1;
		EncryptedValues += Add2;
		EncryptedValues ^= Xor2;
		EncryptedValues += Add3;


		FormatString.Format(L"0x%02X, ", EncryptedValues);
		ResultsSting.Append(FormatString);
	}

	ResultsSting.Append(L" 0 }; \r\n\r\n");

	FormatString.Format(L"for (size_t i = 0; i < %d; i++)\r\n", InString.GetLength());
	ResultsSting.Append(FormatString);

	FormatString.Format(L"{\r\n");
	ResultsSting.Append(FormatString);

	FormatString.Format(L"\tas_%ls[i] -= 0x%02X;\r\n", InString.GetBuffer(), Add3);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tas_%ls[i] ^= 0x%02X;\r\n", InString.GetBuffer(), Xor2);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tas_%ls[i] -= 0x%02X;\r\n", InString.GetBuffer(), Add2);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tas_%ls[i] ^= 0x%02X;\r\n", InString.GetBuffer(), Xor1);
	ResultsSting.Append(FormatString);
	FormatString.Format(L"\tas_%ls[i] -= 0x%02X;\r\n", InString.GetBuffer(), Add1);
	ResultsSting.Append(FormatString);

	FormatString.Format(L"};\r\n\r\n");
	ResultsSting.Append(FormatString);

	m_Edit_Results.SetWindowText(ResultsSting);
}


void CStringEncryptionDlg::AnsiAndUniCodeToString(CString InString)
{
	CString AnsiString;
	CString UniCodeString;
	CString FormatString;
	FormatString.Format(L"char as_%ls[] = { ", InString.GetBuffer());
	AnsiString.Append(FormatString);
	for (size_t i = 0; i < InString.GetLength(); i++)
	{
		FormatString.Format(L"'%lc', ", *(InString.GetBuffer() + i));
		AnsiString.Append(FormatString);
	}
	AnsiString.Append(L"0 };");
	m_Edit_Ansi.SetWindowText(AnsiString);

	FormatString.Format(L"wchar_t ws_%ls[] = { ", InString.GetBuffer());
	UniCodeString.Append(FormatString);
	for (size_t i = 0; i < InString.GetLength(); i++)
	{
		FormatString.Format(L"L'%lc', ", *(InString.GetBuffer() + i));
		UniCodeString.Append(FormatString);
	}
	UniCodeString.Append(L"0 };");
	m_Edit_UniCode.SetWindowText(UniCodeString);
}

int CStringEncryptionDlg::GetRand(int MIN, int MAX)
{
	int max = RAND_MAX;
	return (int)(rand() * (MAX - MIN) / max + MIN);
}



void CStringEncryptionDlg::OnOK()
{
	// TODO: 在此添加专用代码和/或调用基类

	//CDialogEx::OnOK();
}


BOOL CStringEncryptionDlg::PreTranslateMessage(MSG* pMsg)
{
	// TODO: 在此添加专用代码和/或调用基类
	if (pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_ESCAPE)
	{
		pMsg->wParam = VK_RETURN;																  
	}

	return CDialogEx::PreTranslateMessage(pMsg);
}
